@echo off
echo ========================================
echo 乳腺癌预测系统 - RAG功能安装脚本
echo ========================================
echo.

echo 正在检查Python版本...
python --version
if %errorlevel% neq 0 (
    echo 错误: 未找到Python，请先安装Python 3.7或更高版本
    pause
    exit /b 1
)

echo.
echo 正在安装RAG系统依赖...
echo 这可能需要几分钟时间，请耐心等待...
echo.

pip install sentence-transformers>=2.2.0
if %errorlevel% neq 0 (
    echo 警告: sentence-transformers安装失败
)

pip install torch>=1.9.0
if %errorlevel% neq 0 (
    echo 警告: torch安装失败
)

pip install transformers>=4.20.0
if %errorlevel% neq 0 (
    echo 警告: transformers安装失败
)

echo.
echo 正在安装其他依赖...
pip install -r requirements.txt

echo.
echo ========================================
echo 安装完成！
echo ========================================
echo.
echo RAG功能已启用，您现在可以：
echo 1. 使用智能问答功能
echo 2. 管理知识库（管理员权限）
echo 3. 享受更丰富的医学知识服务
echo.
echo 请运行 start.bat 启动系统
echo.
pause
